from pytorch_pretrained_bert import BertModel, BertTokenizer
import numpy as np
import torch

tokenizer = BertTokenizer.from_pretrained('F:/archive/bert-base-uncased-vocab.txt')
bert = BertModel.from_pretrained('F:/archive/bert-base-uncased/bert-base-uncased')
s = 'you are some the one of bitch'
tokens = tokenizer.tokenize(s)

ids = torch.tensor([tokenizer.convert_tokens_to_ids(tokens)])


result = bert(ids, output_all_encoded_layers=True)
print(result.shape)